package com.glodon.paas.account.util.captcha1;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

import javax.imageio.ImageIO;
/**
 * 验证码图片生成器
 * 
 * @author 
 *
 */
public class CaptchaImageGenerator {
	// 字体
	private String fontName="Arial";
	// 字型
	private int fontStyle = Font.PLAIN;
	
	// 图片的宽度。
	private int width = 160;
	// 图片的高度。
	private int height = 40;
	// 验证码字符个数
	private int codeCount = 5;
	// 验证码干扰线数
	private int lineCount = 150;
	// 验证码
	private String code = null;
	// 验证码图片Buffer
	private BufferedImage buffImg=null;
	// 去掉了容易误识别字符:0O1lijvy
	private char[] codeSequence = "123456789ABCDEFGHJKLMNPRSTUVXYZ".toCharArray();

	public  CaptchaImageGenerator() {		
		this.createCode();
	}

	/**
	 * 
	 * @param width 图片宽
	 * @param height 图片高
	 */
	public  CaptchaImageGenerator(int width,int height) {
		this.width=width;
		this.height=height;
		this.createCode();
	}
	/**
	 * 
	 * @param width 图片宽
	 * @param height 图片高
	 * @param codeCount 字符个数
	 * @param lineCount 干扰线条数
	 */
	public  CaptchaImageGenerator(int width,int height,int codeCount,int lineCount,String fontName,int fontStyle) {
		this.width=width;
		this.height=height;
		this.codeCount=codeCount;
		this.lineCount=lineCount;
		this.fontName=fontName;
		this.fontStyle=fontStyle;
		this.createCode();
	}
	
	public void createCode() {
		int x = 0,fontHeight=0,codeY=0;
		int red = 0, green = 0, blue = 0;
		
		x = width / (codeCount + 1);//每个字符的宽度
		fontHeight = height - 2;//字体的高度
		codeY = height - 4;
		
		// 图像buffer
		buffImg = new BufferedImage(width, height,BufferedImage.TYPE_INT_RGB);
		Graphics2D g = buffImg.createGraphics();
		// 生成随机数
		Random random = new Random();
		// 将图像填充为白色
		g.setColor(Color.WHITE);
		g.fillRect(0, 0, width, height);
		// 创建字体		
		Font font = FontCache.getFont(fontName, fontStyle, fontHeight);
		g.setFont(font);
		for (int i = 0; i < lineCount; i++) {
			int xs = random.nextInt(width);
			int ys = random.nextInt(height);
			int xe = xs+random.nextInt(width/8);
			int ye = ys+random.nextInt(height/8);
			red = random.nextInt(255);
			green = random.nextInt(255);
			blue = random.nextInt(255);
			g.setColor(new Color(red, green, blue));
			g.drawLine(xs, ys, xe, ye);
		}
		
		// randomCode记录随机产生的验证码
		StringBuffer randomCode = new StringBuffer();
		// 随机产生codeCount个字符的验证码。
		for (int i = 0; i < codeCount; i++) {
			String strRand = String.valueOf(codeSequence[random.nextInt(codeSequence.length)]);
			// 产生随机的颜色值，让输出的每个字符的颜色值都将不同。
			//red = random.nextInt(255);
			//green = random.nextInt(255);
			//blue = random.nextInt(255);
			//g.setColor(new Color(red, green, blue));
			g.setColor(Color.GRAY);
			g.drawString(strRand, i*x+x/2, codeY);
			// 将产生的四个随机数组合在一起。
			randomCode.append(strRand);
		}
		code = randomCode.toString();		
	}
	
	//写文件
	public void write(String path) throws IOException {
		OutputStream sos = new FileOutputStream(path);
			this.write(sos);
	}
	
	public void write(OutputStream sos) throws IOException {
			ImageIO.write(buffImg, "png", sos);
	}
	
	public BufferedImage getBuffImg() {
		return buffImg;
	}
	
	public String getCode() {
		return code;
	}
}
